package com.nuyoah.controller;


import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.nuyoah.common.dto.PassDto;
import com.nuyoah.common.lang.Const;
import com.nuyoah.common.lang.Result;
import com.nuyoah.entity.Role;
import com.nuyoah.entity.User;
import com.nuyoah.entity.UserRole;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.Assert;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.security.Principal;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 用户相关接口
 *
 * @author xiaoming
 * @since 2024-03-26
 */
@RestController
@RequestMapping("/sys/user")
public class UserController extends BaseController {

    @Autowired
    BCryptPasswordEncoder passwordEncoder;

    /**
     * 根据用户ID获取用户信息。
     * @param id 用户ID
     * @return 用户信息结果对象
     */
    @GetMapping("/info/{id}")
    @PreAuthorize("hasAuthority('sys:user:list')")
    public Result info(@PathVariable("id") Long id) {
        // 获取用户信息并校验其存在性
        User user = userService.getById(id);
        Assert.notNull(user, "找不到该管理员");
        // 获取用户角色
        List<Role> roles = roleService.listRolesByUserId(id);

        user.setRoles(roles);
        return Result.succ(user);
    }

    /**
     * 分页获取用户列表。
     * @param username 用户名查询条件
     * @return 用户列表结果对象
     */
    @GetMapping("/list")
    @PreAuthorize("hasAuthority('sys:user:list')")
    public Result list(String username) {
        // 分页查询用户列表并为每个用户加载角色信息
        Page<User> pageData = userService.page(getPage(), new QueryWrapper<User>()
                .like(StrUtil.isNotBlank(username), "username", username));

        pageData.getRecords().forEach(u -> {
            u.setRoles(roleService.listRolesByUserId(u.getId()));
        });

        return Result.succ(pageData);
    }

    /**
     * 保存新用户信息。
     * @param sysUser 待保存的用户对象
     * @return 保存结果对象
     */
    @PostMapping("/save")
    @PreAuthorize("hasAuthority('sys:user:save')")
    public Result save(@Validated @RequestBody User sysUser) {
        // 初始化用户信息并保存
        sysUser.setCreated(LocalDateTime.now());
        sysUser.setStatus(Const.STATUS_ON);

        // 设置默认密码
        String password = passwordEncoder.encode(Const.DEFULT_PASSWORD);
        sysUser.setPassword(password);

        // 设置默认头像
        sysUser.setAvatar(Const.DEFULT_AVATAR);

        userService.save(sysUser);
        return Result.succ(sysUser);
    }

    /**
     * 更新用户信息。
     * @param oldUser 待更新的用户对象
     * @return 更新结果对象
     */
    @PostMapping("/update")
    @PreAuthorize("hasAuthority('sys:user:update')")
    public Result update(@Validated @RequestBody User oldUser) {

        User newUser = new User();
        newUser = oldUser;
        newUser.setPassword(passwordEncoder.encode(oldUser.getPassword()));

        // 更新用户信息
        newUser.setUpdated(LocalDateTime.now());

        userService.updateById(newUser);
        return Result.succ(newUser);
    }

    /**
     * 批量删除用户及其关联角色。
     * @param ids 待删除用户ID数组
     * @return 删除结果对象
     */
    @Transactional
    @PostMapping("/delete")
    @PreAuthorize("hasAuthority('sys:user:delete')")
    public Result delete(@RequestBody Long[] ids) {
        // 删除用户及关联角色
        userService.removeByIds(Arrays.asList(ids));
        userRoleService.remove(new QueryWrapper<UserRole>().in("user_id", ids));

        return Result.succ("");
    }

    /**
     * 设置用户角色。
     * @param userId 用户ID
     * @param roleIds 角色ID数组
     * @return 设置结果对象
     */
    @Transactional
    @PostMapping("/role/{userId}")
    @PreAuthorize("hasAuthority('sys:user:role')")
    public Result rolePerm(@PathVariable("userId") Long userId, @RequestBody Long[] roleIds) {
        // 清除旧角色并设置新角色
        List<UserRole> userRoles = new ArrayList<>();

        Arrays.stream(roleIds).forEach(r -> {
            UserRole userRole = new UserRole();
            userRole.setRoleId(r);
            userRole.setUserId(userId);

            userRoles.add(userRole);
        });

        userRoleService.remove(new QueryWrapper<UserRole>().eq("user_id", userId));
        userRoleService.saveBatch(userRoles);

        // 删除缓存中的用户权限信息
        User user = userService.getById(userId);
        userService.clearUserAuthorityInfo(user.getUsername());

        return Result.succ("");
    }

    /**
     * 重置用户密码。
     * @param userId 用户ID
     * @return 重置结果对象
     */
    @PostMapping("/repass")
    @PreAuthorize("hasAuthority('sys:user:repass')")
    public Result repass(@RequestBody Long userId) {
        // 重置密码并更新时间
        User user = userService.getById(userId);

        user.setPassword(passwordEncoder.encode(Const.DEFULT_PASSWORD));
        user.setUpdated(LocalDateTime.now());

        userService.updateById(user);
        return Result.succ("");
    }

    /**
     * 更新用户密码。
     * @param passDto 包含当前密码和新密码的DTO对象
     * @param principal 当前登录用户信息
     * @return 更新结果对象
     */
    @PostMapping("/updatePass")
    public Result updatePass(@Validated @RequestBody PassDto passDto, Principal principal) {
        // 校验旧密码并更新密码
        User sysUser = userService.getByUsername(principal.getName());

        boolean matches = passwordEncoder.matches(passDto.getCurrentPass(), sysUser.getPassword());
        if (!matches) {
            return Result.fail("旧密码不正确");
        }

        sysUser.setPassword(passwordEncoder.encode(passDto.getPassword()));
        sysUser.setUpdated(LocalDateTime.now());

        userService.updateById(sysUser);
        return Result.succ("");
    }

    /**
     * 普通用户注册接口。
     *
     * @param oldUer 请求体中的用户信息，需要验证其合法性。
     * @return 注册结果，成功返回包含用户信息的结果对象，失败则返回相应错误信息。
     */
    @PostMapping("/register")
    public Result register(@Validated @RequestBody User oldUer) {
        // 创建新的用户实例并初始化
        User user = new User();
        user = oldUer;
        // 对用户密码进行加密处理
        user.setPassword(passwordEncoder.encode(oldUer.getPassword()));
        // 设置用户创建时间
        user.setCreated(LocalDateTime.now());
        // 设置用户初始状态为禁用
        user.setStatus(Const.STATUS_OFF);
        // 保存用户信息
        userService.save(user);

        // 为当前注册用户添加默认角色
        UserRole userRole = new UserRole();
        userRole.setUserId(user.getId());
        userRole.setRoleId(3L); // 默认角色ID为3
        userRoleService.save(userRole);

        // 返回注册成功的用户信息
        return Result.succ(user);
    }


}
